"""
TPshop的登录接口封装

1、获取验证码  http://localhost:8080/index.php?m=Home&c=User&a=verify
2、登录  http://localhost:8080/index.php?m=Home&c=User&a=do_login
"""

import requests

# 定义接口的类
class LoginAPI(object):
    # 每次都要使用这两个url地址，可以放在初始化函数中
    def __init__(self):
        self.url_verify = "http://localhost:9096/index.php?m=Home&c=User&a=verify"
        self.url_login = "http://localhost:9096/index.php?m=Home&c=User&a=do_login"

    # 获取验证码的方法：传递的时候需要session
    def get_verify_code(self,session):
        return session.get(url=self.url_verify)

    # 登录方法：传递数值有：session、username、password、verify_code
    def login(self,session,username,password,verify_code):
        login_data = {
            "username":username,
            "password":password,
            "verify_code":verify_code
        }
        return session.post(url=self.url_login,data=login_data)
